import argparse
from format import process_directory
from pathlib import Path

if __name__ == '__main__':
    # 设置命令行参数
    parser = argparse.ArgumentParser(description="汇总材料清册")
    parser.add_argument(
        "-i", "--input", 
        required=False,
        help="输入文件夹路径(包含.xlsx 和 .xls 文件)"
    )
    parser.add_argument(
        "-o", "--output", 
        default="output_excels",
        help="输出文件夹路径(默认: 当前目录下的output_excels)"
    )

    args = parser.parse_args()

    # 验证输入路径
    if args.input and not Path(args.input).exists():
        print(f"输入路径 {args.input} 不存在!")

    if args.input:
        process_directory(args.input, args.output)
    else:
        process_directory(r'D:\Users\lijinsen\Desktop\qiji_excel', args.output)